import java.util.HashMap;
import java.util.Map;

/*
 * @lc app=leetcode.cn id=525 lang=java
 *
 * [525] 连续数组
 *
 * https://leetcode-cn.com/problems/contiguous-array/description/
 *
 * algorithms
 * Medium (39.85%)
 * Likes:    70
 * Dislikes: 0
 * Total Accepted:    2.7K
 * Total Submissions: 6.7K
 * Testcase Example:  '[0,1]'
 *
 * 给定一个二进制数组, 找到含有相同数量的 0 和 1 的最长连续子数组（的长度）。
 *
 *
 *
 * 示例 1:
 *
 * 输入: [0,1]
 * 输出: 2
 * 说明: [0, 1] 是具有相同数量0和1的最长连续子数组。
 *
 * 示例 2:
 *
 * 输入: [0,1,0]
 * 输出: 2
 * 说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。
 *
 *
 *
 * 注意: 给定的二进制数组的长度不会超过50000。
 *
 */

// @lc code=start
class Solution {
    public int findMaxLength(int[] nums) {
        int len = 0;
        int sum = 0;
        Map<Integer, Integer> map = new HashMap<>();
        for(int i=0; i < nums.length; i++) {
            if (nums[i]==1)
                sum++;
            else
                sum--;
            if(sum == 0)
                len = i+1;
            else {
                if(map.containsKey(sum)) {
                    int value = map.get(sum);
                    len = len > i - value ? len : i - value;
                } else {
                    map.put(sum, i);
                }
            }
        }
        return len;
    }
}
// @lc code=end

